#pragma once//别再弄丢了

#include <iostream>
#include <memory>
#include "udpserver.hpp"//引入通信模块（功能）
#include "Routine.hpp"//引入消息路由功能
#include "threadpool.hpp"


using namespace ThreadPoolModule;

//for test
std::string defaulthandler(const std::string &message)
{
    std::string hello = "hello, ";
    hello += message;
    return hello;
}


using task_t=std::function<void()>;

//端口号以命令行参数的形式获取
int main(int argc, char *argv[])
{
    if(argc != 2)
    {
        std::cerr << "Usage: " << argv[0] << " port" << std::endl;
        return 1;
    }
    uint16_t port = std::stoi(argv[1]);
    Enable_Console_Log_Strategy();

    //1，聊天模块
    //将所有客户端发来的消息转发出去，让所有客户端都可以看到这个消息
    //起到路由消息的功能
    Routine r;  

    //2,线程池
    //获取单例
    auto tp=ThreadPool<task_t>::GetInstance();

    //3,通信模块
    //传入回调方法
    //进行消息的路由处理
    //需要的参数：套接字，消息，发送对象
    std::unique_ptr<udpserver> usvr = std::make_unique<udpserver>(port,[&r,&tp](int sockfd,const std::string& message,InetAddr& addr)
    {
        //参数绑死
        //在线程池中可以通过t()的方式执行
        task_t t=std::bind(&Routine::MessageRoutine,&r,sockfd,message,addr);
        tp->Enque(t);//任务入线程池
    });

    usvr->init();
    usvr->start();

    return 0;
}